/**
 * @file feature.h
 * @author RittoQin (1264145445@qq.com)
 * @brief 
 * @version 1.0
 * @date 2022/4/24
 *
 * @copyright Copyright SCUT RobotLab(c) 2022
 *
 */

#include <opencv2/core.hpp>
#include "Yolo.h"
#include "Parameter.h"

class feature
{
public:
    feature() {
        yolo = std::make_shared<Yolo>();
    };

    std::vector<BoxInfo> infer(cv::Mat &);

    void write(const std::vector<BoxInfo>&);

    std::shared_ptr<Yolo> yolo;

    std::vector<BoxInfo> boxes;
};

using feature_ptr = std::shared_ptr<feature>;